from pymongo import MongoClient
from app.core.config import MONGODB_URI, DATABASE_NAME

client = MongoClient(MONGODB_URI)
db = client[DATABASE_NAME]

class GeneratedPromptCRUD:
    @staticmethod
    async def create_prompt(prompt_data):
        return await db.generation_prompt.insert_one(prompt_data)

    @staticmethod
    async def get_prompt_by_id(prompt_id):
        return await db.generation_prompt.find_one({'_id': prompt_id})

    @staticmethod
    async def update_prompt(prompt_id, update_data):
        return await db.generation_prompt.update_one({'_id': prompt_id}, {'$set': update_data})

    @staticmethod
    async def delete_prompt(prompt_id):
        return await db.generation_prompt.delete_one({'_id': prompt_id})
